export CUDA_VISIBLE_DEVICES=0
export OMP_NUM_THREADS=1

PYTHONPATH="$(dirname $0)/..":"$(dirname $0)/../stable-diffusion":$PYTHONPATH \
python train.py \
    --batch_size 2 \
    --dataset nyudepthv2 \
    --data_path ./data/nyu \
    --weight_decay 0.1 \
    --num_filters 32 32 32 \
    --deconv_kernels 2 2 2 \
    --flip_test \
    --layer_decay 0.9 \
    --drop_path_rate 0.3 \
    ${@:2} \
    --pro_bar True \
    --save_depths_gray \
    --save_depths_color \
    --vit_model google/vit-base-patch16-224 \
    --local_vit_model_path ./vit_models/vit-base-patch16-224 \
    --learning_rate_schedule one_cycle \
    --gradient_accumulation False \
    --epochs 25 \
    --log_dir log_dir \
    --log_in_wandb True \
    --eigen_crop_in_dataloader_itself_for_nyu False \
    --min_depth_eval 1e-3 \
    --max_depth_eval 10 \
    --exp_name training_nyu \
    --no_of_classes 100 \
    --variance_focus 0.85 \
    --cutflip=False \
    --save_last_model